package cn.tedu._05weibo.controller;

import cn.tedu._05weibo.base.response.JsonResult;
import cn.tedu._05weibo.base.response.StatusCode;
import cn.tedu._05weibo.mapper.UserMapper;
import cn.tedu._05weibo.pojo.dto.UserLoginParam;
import cn.tedu._05weibo.pojo.dto.UserRegParam;
import cn.tedu._05weibo.pojo.entity.User;
import cn.tedu._05weibo.pojo.vo.UserVO;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import springfox.documentation.annotations.ApiIgnore;

import javax.servlet.http.HttpSession;
import java.util.Date;

@Slf4j
@RestController
@RequestMapping("/v1/user/")
@Api(tags = "01.用户模块")
public class UserController {
    @Autowired
    private UserMapper userMapper;

    @PostMapping("reg")
    @ApiOperation(value = "注册功能")
    @ApiOperationSupport(order = 10)
    public JsonResult reg(@Validated @RequestBody UserRegParam userRegParam) {
        log.debug("userRegParam: " + userRegParam);
        String username = userRegParam.getUsername();
        int count = userMapper.selectUserByUsername(username);
        if (count > 0) {
            return new JsonResult(StatusCode.USERNAME_EXISTS);
        }

        User user = new User();
        BeanUtils.copyProperties(userRegParam, user);
        user.setCreateTime(new Date());
        log.debug("user: " + user);
        count = userMapper.insertUser(user);
        return count > 0 ? JsonResult.ok() : new JsonResult(StatusCode.OPERATION_FAILED);
    }

    @PostMapping("login")
    @ApiOperation(value = "登录功能")
    @ApiOperationSupport(order = 20)
    public JsonResult login(@RequestBody UserLoginParam userLoginParam, @ApiIgnore HttpSession session) {
        log.debug("userLoginParam: " + userLoginParam);
        UserVO userVO = userMapper.selectUserByUsernameAndPassword(userLoginParam);
        if (userVO == null) {
            return new JsonResult(StatusCode.USERNAME_PASSWORD_ERROR);
        }
        session.setAttribute("user", userVO);
        return JsonResult.ok(userVO);
    }

    @GetMapping("currentUser")
    @ApiOperation(value = "获取当前用户")
    @ApiOperationSupport(order = 30)
    public JsonResult currentUser(@ApiIgnore HttpSession session) {
        UserVO userVO = (UserVO) session.getAttribute("user");
        log.debug("userVO: " + userVO);
        return userVO != null ? JsonResult.ok(userVO) : new JsonResult(StatusCode.OPERATION_FAILED);
    }

    @GetMapping("logout")
    @ApiOperation(value = "退出登录")
    @ApiOperationSupport(order = 40)
    public void logout(@ApiIgnore HttpSession session) {
        session.removeAttribute("user");
    }

}
